package com.zcn.io.common.util;

import cn.hutool.core.collection.CollectionUtil;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.function.Function;

/**
 * @desc:
 * @author: 202438 邱建伟
 * @date: 2025/3/23 14:39
 * @Copyright (c) 2025, 江苏金恒信息科技股份有限公司
 */
public class CollectionUtils extends CollectionUtil {

    /**
     * 根据传入的函数指定的字段对列表进行分组
     * @param list 待分组的列表
     * @param classifier 用于指定分组字段的函数
     * @param <T> 列表元素的类型
     * @param <K> 分组键的类型
     * @return 分组后的 Map
     */
    public static <T, K> Map<K, List<T>> groupBy(List<T> list, Function<? super T, ? extends K> classifier) {
        Map<K, List<T>> result = new HashMap<>();
        for (T item : list) {
            K key = classifier.apply(item);
            result.computeIfAbsent(key, k -> new ArrayList<>()).add(item);
        }
        return result;
    }

}
